More Pythonic way of writing If-Else-Pass